// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "net/base/address_family.h"

#include "net/base/ip_address.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {
namespace {

    TEST(AddressFamilyTest, GetAddressFamily)
    {
        IPAddress address;
        EXPECT_EQ(ADDRESS_FAMILY_UNSPECIFIED, GetAddressFamily(address));
        EXPECT_TRUE(address.AssignFromIPLiteral("192.168.0.1"));
        EXPECT_EQ(ADDRESS_FAMILY_IPV4, GetAddressFamily(address));
        EXPECT_TRUE(address.AssignFromIPLiteral("1:abcd::3:4:ff"));
        EXPECT_EQ(ADDRESS_FAMILY_IPV6, GetAddressFamily(address));
    }

} // namespace
} // namespace net
